package com.example.authority.controller;
import cn.dev33.satoken.stp.StpUtil;
import com.alibaba.excel.EasyExcel;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.ObjectUtil;
import com.alibaba.excel.read.listener.PageReadListener;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.example.authority.common.Result;
import com.example.authority.entity.Userhave;
import com.example.authority.service.UserRoleService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.net.URLEncoder;
import java.util.Arrays;
import java.util.List;


import com.example.authority.service.GoldthingtyService;
import com.example.authority.entity.Goldthingty ;
import com.example.authority.service.GodlthingdeService;
import com.example.authority.entity.Godlthingde ;
import com.example.authority.service.UsergodthingService;
import com.example.authority.entity.Usergodthing;



/**
 * @program: authority-v3.0.7
 * @ClassName:UsergodthingController
 * @description: UsergodthingController前端控制器
 * @author:dyy
 * @Version 3.0
 **/
@Tag(name = "圣遗物 前端控制器")
    @RestController
@RequestMapping("/usergodthing")
    public class UsergodthingController {

@Resource
private UsergodthingService usergodthingService;
@Resource
private GoldthingtyService goldthingtyService;
@Resource
private GodlthingdeService godlthingdeService;
@Resource
private UserRoleService userRoleService;

/**
 * 新增
 * @param usergodthing
 * @return
 */
@Operation(summary = "新增")
@PostMapping
public Result save(@RequestBody Usergodthing usergodthing){
    Object loginId = StpUtil.getLoginId();
    if (loginId instanceof String) {
        usergodthing.setUserid(Long.valueOf((String) loginId));
    } else if (loginId instanceof Long) {
        usergodthing.setUserid((Long) loginId);
    }
        return Result.success(usergodthingService.save(usergodthing));
        }
/**
* 修改
*
* @param usergodthing
* @return
*/
@Operation(summary = "修改")
@PutMapping("/{id}")
public Result update(@PathVariable Long id,@RequestBody Usergodthing usergodthing) {
        return Result.success(usergodthingService.updateById(usergodthing));
        }

/**
 * 查询所有Usergodthing
 * @return
 */
@Operation(summary = "查询所有Usergodthing")
@GetMapping
public Result findAll(){
        return Result.success(usergodthingService.list());
        }
/**
  * 获取单个
  * @param id
  * @return
  */
@Operation(summary = "获取单个")
@GetMapping("/{id}")
public Result findOne(@PathVariable Integer id){
        return Result.success(usergodthingService.getById(id));
        }
/**
 * 分页显示
 * @param
 * @param pageNum
 * @param pageSize
 * @return
 */
@Operation(summary = "分页显示")
@GetMapping("/page")
public Result findPage(
        @RequestParam(defaultValue = "") Long type,
        @RequestParam(defaultValue = "") String part,
        @RequestParam(defaultValue = "") Integer evaluate,
@RequestParam(defaultValue = "1") Integer pageNum,
@RequestParam(defaultValue = "10") Integer pageSize){

        LambdaQueryWrapper<Usergodthing>queryWrapper=new LambdaQueryWrapper();
        if (type!=null&&type!=0) {
            // name 不为空且不是空字符串的处理逻辑
            queryWrapper.eq(Usergodthing::getType,type);
        }
        if (part!=null&& !part.equals("")) {
            // name 不为空且不是空字符串的处理逻辑
            queryWrapper.eq(Usergodthing::getPart,part);
        }
        if (evaluate!=null&&evaluate!=0) {
            // name 不为空且不是空字符串的处理逻辑
            queryWrapper.eq(Usergodthing::getEvaluate,evaluate);
        }

    // 获取当前用户ID
    Long currentUserId = Long.valueOf(StpUtil.getLoginId().toString());
    List<Long> roleIds = userRoleService.getRoleIds(currentUserId, "PC");

    if(roleIds.get(0) !=1){
        // 添加当前用户ID的查询条件
        queryWrapper.eq(Usergodthing::getUserid, currentUserId);
    }

        queryWrapper.orderByDesc(Usergodthing::getId);
        Page<Usergodthing>page=usergodthingService.page(new Page<>(pageNum,pageSize),queryWrapper);
        List<Usergodthing > records = page.getRecords();
         for (Usergodthing usergodthing : records) {
                 Goldthingty goldthingty = goldthingtyService.getById(usergodthing.getType());
                 if (ObjectUtil.isNotEmpty(goldthingty)){
                     usergodthing.setGoldthingty(goldthingty);
                 }
                 Godlthingde godlthingde = godlthingdeService.getById(usergodthing.getPart());
                 if (ObjectUtil.isNotEmpty(godlthingde)){
                     usergodthing.setGodlthingde(godlthingde);
                 }
         }
         page.setRecords(records);
         JSONObject result = new JSONObject();
         result.put("roleId", roleIds.isEmpty() ? null : roleIds.get(0));
         result.put("page", page);
         return Result.success(result);

        }
/**
* 单个删除
* @param id
* @return
*/
@Operation(summary = "单个删除")
@DeleteMapping("/{id}")
public Result delete(@PathVariable Integer id){
        return Result.success(usergodthingService.removeById(id));
        }
/**
* 批量删除
* @param ids
* @return
*/
@Operation(summary = "批量删除")
@DeleteMapping("/batch/{ids}")
@Transactional
public Result deleteByIds(@PathVariable String[]ids){
        return Result.success(usergodthingService.removeByIds(Arrays.asList(ids)));
        }

/**
 * 批量导出
 * 使用的技术为alibaba下面的easyexcel
 * 写数据
 *
 * @param ids
 * @return
 */
@Operation(summary = "批量导出")
@GetMapping("/batch/export/{ids}")
public void exportByIds(@PathVariable String[] ids, HttpServletResponse response) throws IOException {

        List<Usergodthing> list = usergodthingService.listByIds(Arrays.asList(ids));
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("usergodthing导出数据", "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), Usergodthing.class).sheet("sheel1").doWrite(list);

        }

/**
 * 批量导入
 * 使用的技术为alibaba下面的easyexcel
 * 读数据
 *
 * @param
 */
@Operation(summary = "批量导入")
@PostMapping("/batch/upload")
public Result writeExcel(MultipartFile file) throws IOException {
        EasyExcel.read(file.getInputStream(), Usergodthing.class, new PageReadListener<Usergodthing>(dataList -> {
        dataList.forEach(entity -> usergodthingService.save(entity.toBuilder().id(null).build()));
        })).sheet().doRead();
        return Result.success();
        }

        }